Fixed wordcount logic; added more styling