adding condition for lambda=0. Check