Adding testing for speed of code