fixed some small things
changes, hope it doesn't break