Added icon generation - still not working :(