Ok, what I learn and what I do
The early stage I play tight, with a select range: AA, KK, QQ, JJ, AK, AQ, AJ, A10, KQ. Usually like to play in position
The middle stage I open my range to KJ, K10, Q10, QJ, J10, suited connectors, middle pairs and I like to play I little bit more aggressive then on early stage.
Late Stages I like to play with all the cards that I played on early and middle, but ALWAYS in position. You can include Axs at BTN and Cut Off if the table folds until you speak
Good advice for you, do not 3-bet all the time with AK, avoid situations wheres flop are to dry.