Added possibility of passing file names as a param and fixed bug Anna pointed out.
Tiny commit