I have completed the assignment in 2 hours. I could not do the bonus section for accpeting the input from the user as I ran out of time. If i had more time I would also write some more testcases.