add update query
add tests