Updated the input section