refactor unit info gui into its own class