Modify controller to get user input functional -v4