edited main to take two params