Added the connection part