lab01: part II - code cleanup