added menu also tweaked it a bit