Made a few changes

Oh yeah