Done, but with bug in picking word family