Fixat flow controll och städat i koden