Create disk_usage.py

#16 Add disk_usage function