Add test
Remove unused variable
Update test