Uppdaterad kontrollfunktion