Bit of reworking, program compiles now.