improve upon my submission for a1