More display stuff and input handling