Changes to my idiot code

Here it is