Fixed undefined method issue (closes #4)