Add checks for valid input