Adding a few missing use statements