Issue/11

fixes #11 
Yeah this will work for now.