Set size of screen

Added current and old states for input