Besides simplifying the type annotations, can you remove the chained assignments in the first two lines and change the last two to `x = cx + ..`?