fixed tests
added new test
fixed expand function to pass the tests
updated readme