added functionality for work. will move on to something else now