Cleaned up some stuff. Added support for some imm instructions

Next step: add load and store