UPDATE: Forgot to reformat code