Added db connection and button