Made some modifications to database example