added baseline tagger
added utils to use argmax