code clean up a bit