Add check on file size