Cleanup

Removed unnecessary complexity in example