Getting rid of update body in MainState