Main is updated for user inpu