Simple proportional controller added