pretty code fixed