Update if-else statment