Add benchmarking stuff for lab 4