Was missing a system import, and cleaned up some formatting