Some Changes

I i wrote some changes that i think will make your awesome code even better!