Working on register allocation