Trying to get connection class working