adding some of my code to this