updated with my code

I think I got it right