Fixing code formating