Add in Missing Code

Last commit somehow removed code... this puts the code back in.