Add function for input setup