week #11 assignment -  update subclasses of State