Code for Amazon Nex Challenge