Code cleaning return None