nit: use curly brackets even if it's one line