Fixing the repo and adding database code