move it to an inner class