Cleaned up example (fixed types)