refactoring - add Optional