Adding discard button and tweaking some styles.