[#172] Fixed code example and updated to sonata changes