Changed to argparser
Fixed category selection