What I think is probably wrong. I need to go back to my original code