Fixed the problem with selecting units :)
Merge branch 'master' of github.com:justia02/PSI-team-3