Reformatted code to be prettier