Fixes #8 moved waitForInput from keys module to ui class.