Add additional tests ; add connection is the class setup