fixed code, now example to check