Script improvements and usage.

Signed-off-by: Aaron Jacobs <jacobsad@mcmaster.ca>