Add if (but not working)