prettified the code

@abuiles thanks for the help!