Added main to show how input is used