Update kway.py

which takes user input