P2: code cleanup