Code review cleanup

did you add tests for these ability updates?