Add same code in both