add menu state back in... for now