Submission for coursera

Added a few lines to test the functions.