Correct documentation

Fix #4