I wanted to modify something in my code and don't really know how to make this work... my code is