Updated to astropy v0.4