code cleanup
more code cleanup