add className to prototype, remove unwanted code