added if statment