more idiomatic & correct