Add error checking in for disk specification