Fixed doc and added test