ak_visuals: Refactored a bit