Small changes for nicer plotting

In the future, the coefficient should be learned from data size.