Just add if-statements until it works