Messed up register and register_step right now