In my pygame movement system, how do i make it so when I press a key, then press another key so I go diagonally, and then release one of the keys, for it to continue in only