works a bit
added sets to main