added more code to challenge 7