minor update to classes