Added more code, but this doesn't compile and I'm not sure why.