Modified class to decide file content and files