Как в PyTelegramapi настроить кнопки?

Хочу чтобы после нажатия на одну из кнопок предложенных в меню начинало выполняться определённое условие в боте. Задумка такова, что будет две кнопки, после нажатия на одну у пользователя запрашивают 3 переменные и проводят с ними математические операции, после нажатия на другую запрашивают 4 переменные и проводят другие операции. Есть ли что-то, что поможет создать такое условие? Чтобы после нажатия на определённую кнопку выполнялся тот код, который прописан под ней? По отдельности код под кнопками у меня работает, но при попытке запуска одновременно всё встаёт