minor changes
minor
working code yay