Fixed docs error and changed example