Horror
 - add database connect code