Fix/improve __prune__ (fixes #3)