updated code and started challenge 6