Now takes arguments from the command line

Signed-off-by: Thomas Hodgson <thomas.hodgson@gmail.com>