[Fix #44]  Better mobile UI

You should add a Done button on the top right here