And remove all those `if`/`else` statements here.