Reformatted some code to keep consistent with style