Updated classes to work.