Fixed code

I deleted one too many things earlier.