Prettied up if-condition slightly