Refactoring

Changed == to ===