Adding the file size check