Added unit test cases for list comprehensions and fixed a few bugs