Added mock class for diskspace as well as the Storage class