cleaned test
cleaned code