Added Graphics in the stateUI class