Don't need a separate class