Fixed problem with code formating