Fix code formating