lecture 5 fix code formating