This fixes #4

Well this is nice.