State class is now fully functional!