Added the states to MainActivity