Change custom animation example to use new API

This can be merged when 0.5.0 is released