update code for new job class