Bug fix

Improved selection rendering