do you know of any code anywhere that i could add to the above sketch?
There is code posted every single day on the forum that you could add to the sketch. Whether you want to, or not, is a different issue.
What is it that you want the sketch to do that it is not doing? THAT is the code that you need to add.