Closes #7

Signed-off-by: Fadion Dashi <jonidashi@gmail.com>