Update docs and code to version 0.4.0