Solved homework #4 again - removed last line with extra puts