Add use statements

Some imports were missing. This PR adds them.