Changed to function based, added arrow controls

NEED SLEEP!