Could be more clean but overall is ok