Updated the constructor, removed some unnecessary code, finished the else if statement in setCurrentState, and added some comments for readability.