Refactored input a bit.