Edited R file from forked repository.

Changed the R file form the forked git repository with my solution to
assignment 2. I believe it works now :-)