Adding in some line profiling