Add main functionatily