edited main (still missing file i/o and input validation