assignment-1 -- better, faster solution