nitial code and test
updated test slightly