Adding test for lstmap