clean up

cleaning up some code