Made no actual changes to sample.py.. just moved things around
Added final time stamp