Working at #6. Adding migration to cleanup db