clean up code to try to get it to work