Add init, add query
Update tests