Add some extra sanity tests for input
Update and improve example in doc string