updated input code