added test and added todo