Added comments
Working on selection visualization
Shows which entity is selected