Removed test cases and added user input.