fixes to challenge 1