clean up code

highlight not working