Updated AWE code