Added control law. Dunno if it works though.