Remove line added by mistake

Also decided to use match instead of if/else