Using flag package instead of handling args manually