Updates AMI + minor fixes