Started adding DB connection stuff