working on the connection with DB