updated the challenge complete and added some tests
updated the challenge complete and added some tests