updating code based on review comments