Updated test cases and Added Prime Number Logic