Updated docs to 0.4.0 version

Fixed some missing classes