cleanup

I cleaned some code