Profiling for memory usage with memory_profiler