Add feature box code to main project