Completed unit parent-child relationship and added drop to database.py