add appropriate classes to save/draft buttons