Nicer plotting works