Using  variable instead of hardcoded value

Closes #63