added first test and code to pass it