Initial implementation of check_ami.py
Use more idiomatic way
Address the comments
Merge pull request #4 from rail/check_ami

Check AMI