Modifying DB class to accept files