completed #Story18 with this change