added test cases and assertion for new features