Updated input section