So I finally figured out how to get the headlights to stay on for my chosen time limit (20 seconds) once I lock the car.

Turns out you need to press the start/stop button twice so that it goes engine off -> ignition off. Then flash your beams, get out the car and your headlight stays on for the set time.

Does anyone know if there is a way to make it do this without having to flash your beam everytime?