clean up a little bit