Edited the connection, to make it Object Orientated