small changes before handin