added unit test and fixed disk category regex bug