Adding my database code