# Tests for main functionality