Added close button (optional) #13
Some changes in code's style