adding classes to put state data into db