Merge pull request #16 from arjanoosting/fix-annotations

Fix return annotation