Cleaning up and update doc string