Changed setup for diamond structure too.