Små ændringer til controller