Code cleanup
Eensie weensie code change