Modification to code to give additional timing