Change code structure, won't compile