Fixing Issue where script wouldn't work for current directory.

Signed-off-by: Euan T <euantorano@gmail.com>