Added connection to DB