broken, working on legendWidget