add hw3 part3
optionally accept two args
handles both arguments fine but code is messy