missed two return statements
added two examples