just make it a sub class