Added first part of the assignment. Seems to work without feedback