More work on battle stat ui