How can I fulfill the 2nd requirement? Is there any problem with my code?