Add basic input handling