Multi-function pen