Add type annotations so the thing compiles