Added method to get user's input