improved some stuff

I fixed a critical flaw in your stuff.