Add option to change db connection