Added more functionality to State 1
Added more functionality to state 1