Polished src code
Add prompt and quit action