Fixed messed up code