finished off program logic. easy