clean up code - small commit