wrote comments with possible solution