Переключение экрана в Pygame

У меня несколько режимов, которые показывают разные рисунки, фигуры и тд. Их при нажатии клавишы нужно их переключать. Можете подсказать как на одном окне Pygame при нажатии определенной кнопки клавишы менялись режимы? Чтобы было еще понятнее: Первый режим по умолчанию открывает. Нажимается клавиатура переключается на другой режим и работает с ним и тд.