Fixing my bit of the old assignment

added setup for adding to the database