Adding partition check

Works for me. What about putting it in disk file module?