Fixed selecting all elements issue

There are probably a few bugs as I didn't have time to refactor everything, test and let me know.