Rework command to use seperate class to get files. Mock this object in the tests