Clean up

Just a bit of code clean up