cleaned up code on work page