Adding tests upto lesson 10: small modifications