better solution
added diff21 problem
Finished first attempt which passed all tests
Works
One liner solution from CB