Updated code to latest rust nightly