clean up

made it work yo