Add wordcount functionality and styling