added argument to main