Added class to handle files