Tilføjet check i main.m