Add tracking for position of king