Saving caller + callee registers.

Fixed bug in register allocator