Add prune function to main module