Basic version of challenge 6