Use return instead of param for return parameter documentation

Use return instead of param for return parameter documentation #393