solve first test and add one fail