Added in a quick calculation to test code