more progress
added assert to test