Some small fixes

Thank you for this useful example!