Update code

Switch code to proper database.