Should be working Main with bonus feature