Forgot to include 2 lines from original source....