temp solution for first assignment