Added controlls, small fixes.