Fixed code formatiing