Removing leftover code
Merge pull request #8 from hfossli/patch-1

[Fix] Removing leftover code