Fixed some partitioning and caching of rdds as well as adding some additional profiling output