Just some minor changes

For some reason these files weren't being included