All-Star guard expects Golden State to be even better when David Lee returns from injury