added start and end screens
added init/reset function