Improved stoplight detection