Changes made for challenge