polish temporary hack
comment out hack
Fix docstring