Update was calls to astropy 0.4 API

Closes #160