WIP Commit on worker, cleaned up logic around how worker transitions train states