Commented out test functionality and uncommented work functionality.