Add step function in the final pdf

Just for reference, this separate question was followed up on in a separate thread: