Close #4

Closing my first issue