The obvious next step here is to see how well this generalises to arbitrary inputs :)