Merge pull request #1 from SRYUYANG/master

update
Fixed class selection
Merge pull request #3 from maxim123/master

update