Tagit bort if-sats för att bestämma kategori och ersatt med bayesian filtering. Fungerar grymt!