first round for assignment 4
with comments this time