Added Test
small changes.