Was able to add the functionality for flags and validating file extensions.