Add disk space message and test