You probably want to comment this out? It doesn't capture the computation results.

Yes, sorry, forgot to remove that!

collapseAP is a costly function. No need to run it twice. Run it at the beginning and reuse the output here.