refactored yellow square