Add more tests and todo for the imputers